package com.hsp.learn.recursion;

/**
 * @Author Zan
 * @Create 2024/9/24 11:28
 * @ClassName: Queue8
 * @Description : 八皇后问题
 */
public class Queue8 {

    // 定义一个max表示共有多少个皇后
    int max = 8;

    // 定义数组array，保存皇后防止位置的结果，比如 arr = {0, 4, 7, 5, 2, 6, 1, 3} 这个相当于是每一列的位置（所在每一列的某一行上）
    int[] array = new int[max];

    static int count = 0;

    public static void main(String[] args) {
        Queue8 queue8 = new Queue8();
        queue8.check(0); // 把第一个皇后放在第一个位置上（第一列）
        System.out.println(count);
    }

    /**
     * 编写一个方法，放置第n个皇后
     * 特别注意：check是每一次递归时，进入到check中都有一套for循环，因此会有回溯
     */
    private void check(int n) {
        if (n == max) { // n = 8
            // 已经放完了
            print();
            return;
        }
        // 依次放入皇后，并判断是否冲突
        for (int i = 0; i < max; i++) {
            // 先把当前的皇后n放到该行的第一列
            array[n] = i;
            // 判断当放置第n个皇后到i列的时候，是否冲突
            if (judge(n)) {
                // 不冲突
                // 接着放n+1个皇后，即开始递归
                check(n + 1);
            }
            // 如果冲突，就继续for循环执行 array[n] = i; 即将第n个皇后放置在本行的后移一个的位置上
        }
    }

    /**
     * 查看当我们放置第n个皇后，就去检测该皇后是否和前面已经摆放的皇后冲突
     *
     * @param n 表示第n个皇后
     * @return
     */
    private boolean judge(int n) {
        for (int i = 0; i < n; i++) {
            // 1. array[i] == array[n] 表示判断第n个皇后是否和前面的n-1个皇后在同一列
            // 2. Math.abs(array[n] - array[i]) == Math.abs(n - i) 表示判断第n个皇后是否和第i个皇后是否在同一斜线
            // n = 0, 放在第0列第0行，n = 1  放在第1列第1行，即(1, 1)【从0开始】
            // Math.abs：取绝对值
            // Math.abs(1-0) == 1 == Math.abs(1-0) == 1
            // 3. 判断是否在同一行
            if (array[i] == array[n] || Math.abs(array[n] - array[i]) == Math.abs(n - i)) {
                // 说明在同一列 or 在同一斜线
                return false;
            }
        }
        return true;
    }

    // 将皇后摆放的位置打印出来(输出)
    private void print() {
        count++;
        for (int i = 0; i < array.length; i++) {
            System.out.print(array[i] + " ");
        }
        System.out.println();
    }

}
